package com.shm.leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * 216. 组合总和 III
 * 找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。
 * <p>
 * 说明：
 * <p>
 * 所有数字都是正整数。
 * 解集不能包含重复的组合。
 * 示例 1:
 * <p>
 * 输入: k = 3, n = 7
 * 输出: [[1,2,4]]
 * 示例 2:
 * <p>
 * 输入: k = 3, n = 9
 * 输出: [[1,2,6], [1,3,5], [2,3,4]]
 *
 * @author SHM
 */
public class CombinationSum3 {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> temp = new ArrayList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        dfs(k, n, 1);
        return res;
    }

    public void dfs(int k, int n, int start) {
        if (k == 0 && n == 0) {
            res.add(new ArrayList<>(temp));
            return;
        }
        if (start == 10) {
            return;
        }
        if (k == 0) {
            return;
        }
        if (n < start) {
            return;
        }
        temp.add(start);
        dfs(k - 1, n - start, start + 1);
        temp.remove(temp.size() - 1);
        dfs(k, n, start + 1);
    }

    public void dfs_2(int k, int n, int start) {
        if (k == 0 && n == 0) {
            res.add(new ArrayList<>(temp));
            return;
        }
        if (start == 10) {
            return;
        }
        if (k == 0) {
            return;
        }
        if (n < start) {
            return;
        }
        for (int i = start; i < 10; i++) {
            temp.add(i);
            dfs_2(k - 1, n - i, i + 1);
            temp.remove(temp.size() - 1);
        }
    }
}
